package com.nut.course.util;

public class InviteCodeUtils {

    /**
     * 自定义进制(0,1没有加入,容易与o,l混淆)，数组顺序可进行调整增加反推难度，A用来补位因此此数组不包含A，共31个字符。
     */
    private static final char[] BASE = new char[]{
            'H', 'V', 'E', '8', 'S', '2', 'D', 'Z', 'X', '9', 'C', '7', 'P', '5', 'I', 'K', '3', 'M', 'J', 'U',
            'F', 'R', '4', 'W', 'Y', 'L', 'A', 'T', 'N', '6', 'B', 'G', 'Q'};

    /**
     * 进制长度
     */
    private static final int BIN_LEN = BASE.length;

    /**
     * 生成邀请码最小长度
     */
    private static final int CODE_LEN = 6;

    /**
     * ID转换为邀请码
     *
     * @param id
     * @return
     */
    public static String idToCode(Long id) {
        char[] buf = new char[BIN_LEN];
        int charPos = BIN_LEN;

        // 当id除以数组长度结果大于0，则进行取模操作，并以取模的值作为数组的坐标获得对应的字符
        while (id / BIN_LEN > 0) {
            int index = (int) (id % BIN_LEN);
            buf[--charPos] = BASE[index];
            id /= BIN_LEN;
        }

        buf[--charPos] = BASE[(int) (id % BIN_LEN)];
        // 将字符数组转化为字符串
        String result = new String(buf, charPos, BIN_LEN - charPos);

        // 长度不足指定长度则随机补全
//        int len = result.length();
//        if (len < CODE_LEN) {
//            StringBuilder sb = new StringBuilder();
//            sb.append(SUFFIX_CHAR);
//            Random random = new Random();
//            // 去除SUFFIX_CHAR本身占位之后需要补齐的位数
//            for (int i = 0; i < CODE_LEN - len - 1; i++) {
//                sb.append(BASE[random.nextInt(BIN_LEN)]);
//            }
//
//            result += sb.toString();
//        }

        return result;
    }

    /**
     * 邀请码解析出ID<br/>
     * 基本操作思路恰好与idToCode反向操作。
     *
     * @param code
     * @return
     */
    public static Long codeToId(String code) {
        char[] charArray = code.toCharArray();
        long result = 0L;
        for (int i = 0; i < charArray.length; i++) {
            int index = 0;
            for (int j = 0; j < BIN_LEN; j++) {
                if (charArray[i] == BASE[j]) {
                    index = j;
                    break;
                }
            }

//            if (charArray[i] == SUFFIX_CHAR) {
//                break;
//            }

            if (i > 0) {
                result = result * BIN_LEN + index;
            } else {
                result = index;
            }
        }

        return result;

    }

    public static void main(String[] args) {

        Long[] ids = new Long[]{20062330000001L, 20062330000002L, 20062330000003L, 20062330000004L, 20062330000005L, 20062330000006L, 20062330000007L};

        for (Long id : ids) {
            String code = idToCode(id);
            System.out.println(code);
            Long ID = codeToId(code);
            System.out.println(ID);
            System.out.println("----------");
        }

//        String code = idToCode(20062330000012L);
//        System.out.println(code);
//        Long id = codeToId(code);
//        System.out.println(id);
    }

}
